package com.ruoyi.system.service.impl;

import java.util.Date;
import java.util.List;

import com.ruoyi.common.utils.FastDFSUtil;
import com.ruoyi.common.utils.MD5Util;
import io.netty.util.internal.StringUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.ruoyi.system.mapper.TFileMapper;
import com.ruoyi.system.domain.TFile;
import com.ruoyi.system.service.ITFileService;
import org.springframework.web.multipart.MultipartFile;

/**
 * 文件管理Service业务层处理
 *
 * @author Bandit
 * @date 2022-11-06
 */
@Service
public class TFileServiceImpl implements ITFileService
{
    @Autowired
    private TFileMapper tFileMapper;

    @Autowired
    private FastDFSUtil fastDFSUtil;

    /**
     * 分片上传
     * @param slice 文件分片
     * @param fileMD5 文件MD5
     * @param sliceNo 文件序号
     * @param totalSliceNo 分片总数
     * @return url
     * @throws Exception
     */
    public String uploadFileBySlices(MultipartFile slice,
                                     String fileMD5,
                                     Integer sliceNo,
                                     Integer totalSliceNo) throws Exception {
        // 秒传, 看看数据库是否存在文件的md5
        TFile dbFileMD5 = tFileMapper.getFileByMD5(fileMD5);
        if(dbFileMD5 != null){
            return dbFileMD5.getUrl();
        }
        // url只会在所有分片上传完成之后才返回地址， 否则就是空字符串
        String url = fastDFSUtil.uploadFileBySlices(slice, fileMD5, sliceNo, totalSliceNo);
        if(!StringUtil.isNullOrEmpty(url)){
            dbFileMD5 = new TFile();
            dbFileMD5.setCreateTime(new Date());
            dbFileMD5.setMd5(fileMD5);
            dbFileMD5.setUrl(url);
            dbFileMD5.setFileName(fastDFSUtil.getFileName(slice));
            dbFileMD5.setType(fastDFSUtil.getFileType(slice));
            tFileMapper.insertTFile(dbFileMD5);
        }
        return url;
    }

    public String uploadFile(MultipartFile file, String fileMD5) throws Exception {
        // 秒传, 看看数据库是否存在文件的md5
        TFile dbFileMD5 = tFileMapper.getFileByMD5(fileMD5);
        if(dbFileMD5 != null){
            return dbFileMD5.getUrl();
        }
        String url = fastDFSUtil.uploadCommonFile(file);
        if(!StringUtil.isNullOrEmpty(url)){
            dbFileMD5 = new TFile();
            dbFileMD5.setCreateTime(new Date());
            dbFileMD5.setMd5(fileMD5);
            dbFileMD5.setFileName(fastDFSUtil.getFileName(file));
            dbFileMD5.setUrl(url);
            dbFileMD5.setType(fastDFSUtil.getFileType(file));
            tFileMapper.insertTFile(dbFileMD5);
        }
        return url;
    }

    public String getFileMD5(MultipartFile file) throws Exception {
        return MD5Util.getFileMD5(file);
    }

    public TFile getFileByMd5(String fileMd5) {
        return tFileMapper.getFileByMD5(fileMd5);
    }


    /**
     * 查询文件管理
     *
     * @param id 文件管理主键
     * @return 文件管理
     */
    @Override
    public TFile selectTFileById(Long id)
    {
        return tFileMapper.selectTFileById(id);
    }

    /**
     * 查询文件管理列表
     *
     * @param tFile 文件管理
     * @return 文件管理
     */
    @Override
    public List<TFile> selectTFileList(TFile tFile)
    {
        return tFileMapper.selectTFileList(tFile);
    }

    /**
     * 新增文件管理
     *
     * @param tFile 文件管理
     * @return 结果
     */
    @Override
    public int insertTFile(TFile tFile)
    {
        return tFileMapper.insertTFile(tFile);
    }

    /**
     * 修改文件管理
     *
     * @param tFile 文件管理
     * @return 结果
     */
    @Override
    public int updateTFile(TFile tFile)
    {
        return tFileMapper.updateTFile(tFile);
    }

    /**
     * 批量删除文件管理
     *
     * @param ids 需要删除的文件管理主键
     * @return 结果
     */
    @Override
    public int deleteTFileByIds(Long[] ids)
    {
        for (Long id : ids) {
            TFile tFile = tFileMapper.selectTFileById(id);
            String url = tFile.getUrl();
            fastDFSUtil.deleteFile(fastDFSUtil.getBaseUrl() + url);
        }
        return tFileMapper.deleteTFileByIds(ids);
    }

    /**
     * 删除文件管理信息
     *
     * @param id 文件管理主键
     * @return 结果
     */
    @Override
    public int deleteTFileById(Long id)
    {
        return tFileMapper.deleteTFileById(id);
    }
}
